package com.example.myutil;

import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;



public class ImageUtils {
	private static final Logger LOG = LoggerFactory.getLogger(ImageUtils.class);
	/**
	编码的gif图片数据
	编码的png图片数据
	编码的jpeg图片数据
	编码的icon图片数据 
	 * */
	private static final String IMG_PREV = "data:image/%s;base64,";

	/**
	 * @Description: 将base64编码字符串转换为图片
	 * @Author:
	 * @CreateTime:
	 * @param imgStr
	 *            base64编码字符串
	 * @param path
	 *            图片路径-具体到文件
	 * @return
	 */
	public static boolean generateImage(String imgStr, String path) {
		if (imgStr == null)
			return false;
		BASE64Decoder decoder = new BASE64Decoder();
		try {
			// 解密
			byte[] b = decoder.decodeBuffer(imgStr);
			// 处理数据
			for (int i = 0; i < b.length; ++i) {
				if (b[i] < 0) {
					b[i] += 256;
				}
			}
			OutputStream out = new FileOutputStream(path);
			out.write(b);
			out.flush();
			out.close();
			return true;
		} catch (Exception e) {
			return false;
		}
	}

	/**
	 * @Description: 根据图片地址转换为base64编码字符串
	 * @Author:
	 * @CreateTime:
	 * @return
	 */
	public static String getImageStr(String imgFile) {
		InputStream inputStream = null;
		byte[] data = null;
		try {
			inputStream = new FileInputStream(imgFile);
			data = new byte[inputStream.available()];
			inputStream.read(data);
			inputStream.close();
		} catch (IOException e) {
			LOG.error("",e);
		}
		// 加密
		BASE64Encoder encoder = new BASE64Encoder();
		return encoder.encode(data);
	}

	public static String getImageStr(String imgFile, String suffix) {
		InputStream inputStream = null;
		byte[] data = null;
		StringBuffer sb = new StringBuffer();
		try {
			String prefix = null;
			if("png".equals(suffix)){
				prefix = String.format(IMG_PREV, "png");
			}else if("gif".equals(suffix)){
				prefix = String.format(IMG_PREV, "gif");
			}else if("icon".equals(suffix)){
				prefix = String.format(IMG_PREV, "x-icon");
			}else{
				prefix = String.format(IMG_PREV, "jpeg");
			}
			if(prefix != null){
				sb.append(prefix);
			}
			inputStream = new FileInputStream(imgFile);
			data = new byte[inputStream.available()];
			inputStream.read(data);
			inputStream.close();
		} catch (IOException e) {
			LOG.error("",e);
		}
		// 加密
		BASE64Encoder encoder = new BASE64Encoder();
		sb.append(encoder.encode(data));
		return sb.toString();
	}

}
